import json


class Config:
	def __init__(self):
		self.root_path = r'F:\GroupFive\TouManFen'
		# 原始数据路径
		self.train_datapath = self.root_path + r'\01-data\train.csv'
		self.test_datapath = self.root_path + r'\01-data\test.csv'
		self.dev_datapath = self.root_path + r'\01-data\dev.csv'
		# 类别文档
		self.class_doc_path = self.root_path + r"\01-data\class.json"

		# 字符级别fasttext
		self.processed_char_train_datapath = self.root_path + r'\03-fasttext\final_data\processed_char_train.txt'
		self.processed_char_test_datapath = self.root_path + r'\03-fasttext\final_data\processed_char_test.txt'
		self.processed_char_dev_datapath = self.root_path + r'\03-fasttext\final_data\processed_char_dev.txt'

		# 词级别fasttext
		self.processed_word_train_datapath = self.root_path + r'\03-fasttext\final_data\processed_word_train.txt'
		self.processed_word_test_datapath = self.root_path + r'\03-fasttext\final_data\processed_word_test.txt'
		self.processed_word_dev_datapath = self.root_path + r'\03-fasttext\final_data\processed_word_dev.txt'

		# 词级别模型保存路径
		self.ft_model_save_path = self.root_path + r'\03-fasttext\save_models'

		# 类别字典
		with open(self.class_doc_path, 'r', encoding='utf-8') as f:
			data = json.load(f)['label_to_idx']
		self.id2class_dict = {v: k for k, v in data.items()}


if __name__ == '__main__':
	config = Config()
	print(config.id2class_dict)

